'''
Create PT Practice #5
Programmers:
This program will give the user a choice between
   the heartbeat and the billboard
'''
from codex import *
from time import sleep


my_list = ["Ahoy", GREEN, pics.HAPPY, pics.SAD, RED, pics.SURPRISED, 
          pics.ASLEEP, BLUE, pics.HEART, pics.TIARA, pics.TSHIRT,
          ]
choice = 0
LAST_INDEX = len(my_list) - 1
delay = 1.0
my_choice = 0


def display_selection(my_choice):
    global delay, choice
    if my_choice == 1:
        # heartbeat - Keep displaying beats until BTN_D
        while True:
            display.show(pics.HEART)
            sleep(delay)
            display.show(pics.HEART_SMALL)
            sleep(delay)


            if buttons.was_pressed(BTN_A):
                delay = delay + 0.2


            if buttons.was_pressed(BTN_B):
                if delay > 0.2:
                    delay = delay - 0.2
            
            if buttons.was_pressed(BTN_D):
                intro()
                break
    elif my_choice == 2:
        # billboard -- read buttons and show images
        while True:
            my_image = my_list[choice]


            if type(my_image) == tuple:
                display.fill(my_image)
            else:
                display.show(my_image)


            if buttons.was_pressed(BTN_R):
                choice = choice + 1
                if choice > LAST_INDEX:
                    choice = 0


            if buttons.was_pressed(BTN_L):
                choice = choice - 1
                if choice < 0:
                    choice = LAST_INDEX
                    
            if buttons.was_pressed(BTN_D):
                intro()
                break


def intro():
    global my_choice
    display.clear()
    display.print("Press A: Heartbeat")
    display.print("Press B: Billboard")
    display.print("Press D: Quit")
    my_choice = 0


# Main Program
intro()
while True:
    display_selection(my_choice)


    if buttons.was_pressed(BTN_A):
        my_choice = 1


    if buttons.was_pressed(BTN_B):
        my_choice = 2


    if buttons.was_pressed(BTN_D):
        break


# Ending message
display.clear()
display.print("End")